#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
   @Author: 叶建伟
   @Email: yejw46507@hundsun.com
   @Time: 2022/6/8 9:40
   @File: 核查场景新增_多客户含有相同信息.py
"""
import time

from 基础公共脚本.步骤检查_hui2 import check_table,check_tips_msg
from 基础公共脚本.公共方法_hui2 import open_menu,FE
from 基础公共脚本.common_util import auto_fill_form,click_submit
from 数据核查.testUnit.check_scene_util import scene_check,scene_input

# @data_processing
def check_scene_add_sameInfo(input_data=None, ui_expect=None):
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '数据核查/核查场景'})
    FE.button("新增场景").click()
    time.sleep(1)
    add_sameInfo(input_data)
    click_submit()
    # 检查查询结果
    check_table(input_data=None, ui_expect=ui_expect)

def check_scene_add_sameInfo_contrary(input_data=None, ui_expect=None):
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '数据核查/核查场景'})
    FE.button("新增场景").click()
    time.sleep(1)
    add_sameInfo(input_data)
    click_submit()
    # 检查查询结果
    check_tips_msg(ui_expect=ui_expect)

def add_sameInfo(input_data):
    # 表单填入
    text = None
    temp = None
    if input_data.get("相同信息名称"):
        text = input_data.pop("相同信息名称")
    if input_data.get("相同客户数量"):
        temp = input_data.pop("相同客户数量")
    auto_fill_form("场景类型", input_data)
    FE.button(idname={'xpath': '//label[contains(text(),"相同信息名称")]'}).click()
    scene_check("相同信息名称", text)
    print(temp)
    if temp.find("&"):
        tempList = temp.split("&")
        if tempList[0]:
            scene_check("相同客户数量", tempList[0])
        if tempList[1]:
            scene_input("相同客户数量", None, tempList[1])